#ifndef DATABASE_H
#define DATABASE_H

#include <QObject>
#include <QSqlDatabase>
#include <QSqlQuery>
#include <QSqlError>
#include <QDateTime>
#include <QDebug>

class Database : public QObject
{
    Q_OBJECT

public:
    explicit Database(QObject *parent = nullptr);
    ~Database();

    // 初始化数据库
    bool initDatabase();
    
    // 保存聊天消息
    bool saveMessage(const QString &sender, const QString &message, const QString &timestamp);
    
    // 加载历史消息
    QStringList loadHistory(int limit = 100);
    
    // 清空聊天记录
    bool clearHistory();

private:
    QSqlDatabase db;
    bool createTables();
};

#endif // DATABASE_H
